How to find square of matrix